package cn.itcast.star.graph.comfyui.client.handler;

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

/**
 * ComfyUIMessageHandler类用于处理WebSocket消息
 * 该类继承自TextWebSocketHandler，专门用于处理ComfyUI相关的WebSocket通信
 */
public class ComfyUIMessageHandler extends TextWebSocketHandler {

    /**
     * 在WebSocket连接建立后执行的回调方法
     *
     * @param session WebSocket会话对象，包含连接的相关信息
     * @throws Exception 当连接建立过程中发生异常时抛出
     */
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("来自于" + session.getId() + "的连接成功=============");
    }

    /**
     * 处理WebSocket文本消息
     *
     * @param session WebSocket会话对象，用于与客户端进行通信
     * @param message 接收到的文本消息对象，包含客户端发送的文本数据
     * @throws Exception 当消息处理过程中发生错误时抛出异常
     */
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        System.out.println("接受到消息：" + message.getPayload());
    }

}

